package Top_Interview_Questions_Review._012Backtracking;

import java.util.ArrayList;
import java.util.List;

/**
 * @Author: 吕庆龙
 * @Date: 2020/3/6 15:27
 * <p>
 * https://leetcode-cn.com/problems/palindrome-partitioning/solution/hui-su-you-hua-jia-liao-dong-tai-gui-hua-by-liweiw/
 */
public class _0131 {

    public static void main(String[] args) {
        _0131 test = new _0131();
        test.partition("aab");
    }


    /*------------------------------------递归+回溯--------------------------------------------*/

    List<List<String>> res = new ArrayList<>();

    public List<List<String>> partition(String s) {
        int len = s.length();

        if (len == 0) {
            return res;
        }

        List<String> list = new ArrayList<>();
        backtracking(s, 0, len, list);
        return res;
    }

    /**
     * @param s
     * @param start 起始字符的索引
     * @param len   字符串 s 的长度，可以设置为全局变量
     * @param path  记录从根结点到叶子结点的路径
     *
     */
    private void backtracking(String s, int start, int len, List<String> path) {
        if (start == len) {
            res.add(new ArrayList<>(path));
            return;
        }

        for (int i = start; i < len; i++) {

            // 因为截取字符串是消耗性能的，因此，采用传子串索引的方式判断一个子串是否是回文子串
            // 不是的话，剪枝
            if (!checkPalindrome(s, start, i)) {
                continue;
            }
            path.add(s.substring(start, i + 1));

            backtracking(s, i + 1, len, path);
            //回溯,因为path公用一份
            path.remove(path.size() - 1);
        }
    }

    /**
     * 这一步的时间复杂度是 O(N)，因此，可以采用动态规划先把回文子串的结果记录在一个表格里
     *
     * @param str
     * @param left  子串的左边界，可以取到
     * @param right 子串的右边界，可以取到
     * @return
     */
    private boolean checkPalindrome(String str, int left, int right) {
        // 严格小于即可
        while (left < right) {
            if (str.charAt(left) != str.charAt(right)) {
                return false;
            }
            left++;
            right--;
        }
        return true;
    }

    /*---------------------------------------------------------------------------------------*/

}
